Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Prevent duplicate torch_dtype kwargs #115

Merged
merged 1 commit into from
Mar 24, 2024
Merged

Conversation

yasyf
Copy link
Contributor

@yasyf yasyf commented Mar 22, 2024

Throws an error currently if you specify a torch_dtype in the model_config.

TypeError: transformers.models.auto.auto_factory._BaseAutoModelClass.from_pretrained() got multiple values for keyword argument 'torch_dtype'

cc: @iofu728, @QianhuiWu, @XufangLuo, @mydmdm

Throws an error currently if you specify a `torch_dtype` in the `model_config`.

```pycon
TypeError: transformers.models.auto.auto_factory._BaseAutoModelClass.from_pretrained() got multiple values for keyword argument 'torch_dtype'
```
@iofu728 iofu728 self-requested a review March 24, 2024 08:43
@iofu728 iofu728 self-assigned this Mar 24, 2024
@iofu728 iofu728 added the bug Something isn't working label Mar 24, 2024
@iofu728 iofu728 merged commit 7a74721 into microsoft:main Mar 24, 2024
10 checks passed
@iofu728
Copy link
Contributor

iofu728 commented Mar 24, 2024

Thank you for @yasyf's help. It's merged now.

@yasyf yasyf deleted the patch-1 branch March 25, 2024 00:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants